#pragma once

#include <iostream>
#include <vector>
#include <string>
#include <fstream>
#include <mutex>
#include <unordered_map>
#include <boost/algorithm/string.hpp>
#include "cppjieba/Jieba.hpp"
#include "log.hpp"

namespace ns_util
{
    // 读取文件
    class FileUtil
    {
    public:
        static bool ReadFile(const std::string &file_path, std::string *out)
        {
            // 以输入模式(只读)打开file_path
            std::ifstream in(file_path, std::ios::in);
            // 文件没有打开成功
            if(!in.is_open())
            {
                LogMessage(WARNING, "open file %s error", file_path.c_str());
                return false;
            }

            // line用于读取in输入文件流的一行文本
            std::string line;
            // 注意: getline的返回值是一个istream&类型的引用, 而while循环的条件判断需要一个布尔值来决定是否继续循环
            // 但是这么写是可以的, 因为istream类内部重载了强制类型转化运算符, 会将返回的&引用隐式转化为布尔值
            // 当输入流处于有效状态时, 转换后的布尔值为true; 当输入流处于错误状态或到达文件末尾时, 转换后的布尔值为false
            while(std::getline(in, line))
            {
                *out += line;
            }

            return true;
        }
    };

    class StringUtil
    {
    public:
        // 切分
        static void Split(const std::string &target, std::vector<std::string> *out, const std::string &sep)
        {
            // 调用boost库中的split函数实现字符串分割, 将target字符串按照sep分割符分割, 并将结果存储在out中
            boost::split(*out, target, boost::is_any_of(sep), boost::token_compress_on);
        }
    };

    class NonBlockUtil
    {
    public:
        // 设置为非阻塞
        static void SetNonBlock(int fd)
        {
            // 获得描述符
            int fl = fcntl(fd, F_GETFL);
            // 设置为非阻塞, 读取完数据导致阻塞的错误码是EAGAIN和EWOULDBLOCK
            fcntl(fd, F_SETFL, fl | O_NONBLOCK);
        }
    };

    // 定义jieba所需的文件路径
    const char* const DICT_PATH = "./dict/jieba.dict.utf8";       // 中文单词及其词频信息
    const char* const HMM_PATH = "./dict/hmm_model.utf8";         // 用于对未登录词进行分词
    const char* const USER_DICT_PATH = "./dict/user.dict.utf8";   // 用户自定义词典文件路径
    const char* const IDF_PATH = "./dict/idf.utf8";               // 逆文档频率文件路径
    const char* const STOP_WORD_PATH = "./dict/stop_words.utf8";  // 暂停词(停用词)文件路径
    //const char* const STOP_WORD_PATH = "./stop_words.utf8";

    // 设置为单例模式
    class JiebaUtil
    {
    private:
        //static cppjieba::Jieba jieba;
        cppjieba::Jieba jieba;
        std::unordered_map<std::string, bool> stop_words;
    private:
        JiebaUtil(): jieba(DICT_PATH, HMM_PATH, USER_DICT_PATH, IDF_PATH, STOP_WORD_PATH)
        {}
        JiebaUtil(const JiebaUtil&) = delete;
        JiebaUtil &operator=(const JiebaUtil&) = delete;

        static JiebaUtil *instance;
    public:
        // 获得唯一实例
        static JiebaUtil* get_instance()
        {
            static std::mutex mtx;
            if(nullptr == instance)
            {
                mtx.lock();
                if(nullptr == instance)
                {
                    instance = new JiebaUtil();
                    instance->InitJiebaUtil();
                }
                mtx.unlock();
            }

            return instance;
        }

        // 初始化(用于去掉暂停词)
        void InitJiebaUtil()
        {
            // 读取暂停词文件
            std::ifstream in(STOP_WORD_PATH);
            if(!in.is_open())
            {
                LogMessage(FATAL, "load stop words file error");
                return;
            }
            std::string line;
            while(std::getline(in, line))
            {
                // 将暂停词插入到哈希表中
                stop_words.insert({line, true});
            }
            in.close();
        }

        void CutStringHelper(const std::string &src, std::vector<std::string> *out)
        {
            // 用Jieba分词
            jieba.CutForSearch(src, *out);
            // 去掉暂停词
            for(auto iter = out->begin(); iter != out->end(); )
            {
                auto it = stop_words.find(*iter);
                if(it != stop_words.end())
                {
                    // 说明string是暂停词, 需要去掉
                    iter = out->erase(iter);
                }
                else
                {
                    ++iter;
                }
            }
        }

        void CutS(const std::string &src, std::vector<std::string> *out)
        {
            jieba.CutForSearch(src, *out);
        }
        
    public:
        // 用Jieba分割单词, 去掉暂停词
        static void CutStringRemove(const std::string &src, std::vector<std::string> *out)
        {
            ns_util::JiebaUtil::get_instance()->CutStringHelper(src, out);
        }
        
        // 只用Jieba分割单词, 不去掉暂停词
        static void CutString(const std::string &src, std::vector<std::string> *out)
        {
            ns_util::JiebaUtil::get_instance()->CutS(src, out);
        }
    };

    JiebaUtil *JiebaUtil::instance = nullptr;
}
